package br.uff.probabilidade.modelo.arvore;

import java.awt.Color;
import java.awt.Graphics;
import java.awt.Point;

public class Opcao extends No{
	private String nome;
	private float probabilidade;
	private boolean escolhida=false;
	
	public Opcao(String nome, float probabilidade,No pai,Point posicao){
		super(pai,posicao);		
		this.nome = nome;
		this.probabilidade = probabilidade;
	}
	
	public Opcao(String nome, float probabilidade,Point posicao){
		super(posicao);		
		this.nome = nome;
		this.probabilidade = probabilidade;
	}

	public boolean isEscolhida() {
		return escolhida;
	}

	public void setEscolhida(boolean escolhida) {
		this.escolhida = escolhida;
	}

	public String getNome() {
		return nome;
	}

	public float getProbabilidade() {
		return probabilidade;
	}

	@Override
	public void desenha(Graphics g){
		Color oldColor = g.getColor();			
		if(this.isEscolhida())
			g.setColor(Color.RED);
			
		g.drawString(this.nome, super.posicao.x, super.posicao.y);			
		
		g.fillOval(posicao.x, posicao.y, super.raio, super.raio);
		
		g.setColor(oldColor);
		
		if(!this.isFolha())super.desenhaFilhos(g);
		
					
	}
}
